package com.example.leetcode.prcatice;

import java.util.HashMap;
import java.util.Map;

/**
 * 给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。
 *
 *  
 *
 * 示例 1:
 *
 * 输入: nums = [0,1]
 * 输出: 2
 * 说明: [0, 1] 是具有相同数量0和1的最长连续子数组。
 * 示例 2:
 *
 * 输入: nums = [0,1,0]
 * 输出: 2
 * 说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。
 *
 */
public class Test525 {

    /**
     *  将 0看成-1 就是求最大子序列为0的情况
     * @param nums
     * @return
     */
    public int findMaxLength(int[] nums) {
        int n = nums.length;
        int[]sum = new int[n+1];
        for (int i =1 ;i<=n;i++){
            sum[i] = sum[i-1]+(nums[i-1] == 1? 1:-1);
        }
        Map<Integer,Integer> map = new HashMap<>();
        int ans = 0;
        map.put(0,0);
        for (int i =2;i<=n;i++){
            if(!map.containsKey(sum[i-2])) map.put(sum[i-2],i-2);
            if (map.containsKey(sum[i])) ans = Math.max(ans,i-map.get(sum[i]));
        }
        return ans;
    }

    // sum ={0,1,2,1,2,3,2,1}
}
